package com.go.where.manager.util;

import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;

import javax.imageio.ImageIO;

import org.apache.log4j.Logger;

import com.go.where.manager.common.Constants;
import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGImageEncoder;

/**
 * * 缩略图类（通用） 本java类能将jpg、bmp、png、gif图片文件，进行等比或非等比的大小转换。 具体使用方法
 * compressPic(大图片路径,生成小图片路径,大图片文件名,生成小图片文名,生成小图片宽度,生成小图片高度,是否等比缩放(默认为true))
 * 
 * @author java
 * 
 */
public class CompressPicUtil {

	private static final Logger log = Logger.getLogger(CompressPicUtil.class);

	/**
	 * 图片压缩 默认为等比缩放
	 * 
	 * @param saveDir
	 *            保存路径
	 * @param newFileName
	 *            文件名
	 * @param imageFile
	 *            输入文件
	 * @return
	 * @throws IOException
	 */
	public static void compressPicture(String saveDir, String newFileName,
			File imageFile) throws IOException {
		Image img = ImageIO.read(imageFile);
		// 判断图片格式是否正确
		if (img.getWidth(null) == -1) {
			log.info(" can't read,retry!" + "<BR>");
			return;
		} else {
			// 等比缩放
			// 为等比缩放计算输出的图片宽度及高度
			double rate1 = ((double) img.getWidth(null))
					/ (double) Constants.OUTPUT_WIDTH + 0.1;
			double rate2 = ((double) img.getHeight(null))
					/ (double) Constants.OUTPUT_HEIGHT + 0.1;
			// 根据缩放比率大的进行缩放控制
			double rate = rate1 > rate2 ? rate1 : rate2;
			if (rate > 1.2 || rate < 0.85) {
				rate = 0.98;
			}
			int newWidth = (int) (((double) img.getWidth(null)) / rate);
			int newHeight = (int) (((double) img.getHeight(null)) / rate);
			BufferedImage tag = null;
			/*
			 * Image.SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
			 */
			// 缩略图存放路径
			File todir = new File(saveDir + Constants.ZOOM_PATH);
			if (!todir.exists()) {
				todir.mkdir();
			}
			FileOutputStream out = new FileOutputStream(saveDir
					+ Constants.ZOOM_PATH + File.separator + newFileName);
			if (img.getWidth(null) <= Constants.OUTPUT_WIDTH) {
				FileInputStream fis = new FileInputStream(imageFile);
				byte[] buffer = new byte[1024];
				int len = 0;
				while ((len = fis.read(buffer)) > 0) {
					out.write(buffer, 0, len);
				}
				fis.close();
				out.close();
				log.info("原宽度：" + img.getWidth(null) + "原高度："
						+ img.getHeight(null));
				return;
			} else {
				tag = new BufferedImage(newWidth, newHeight,
						BufferedImage.TYPE_INT_RGB);
				log.info("新宽度：" + newWidth + "新高度：" + newHeight);
				tag.getGraphics().drawImage(
						img.getScaledInstance(newWidth, newHeight,
								Image.SCALE_SMOOTH), 0, 0, null);
			}

			// JPEGImageEncoder可适用于其他图片类型的转换
			JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
			encoder.encode(tag);
			out.close();
		}
	}

	public static void compressPicture(String savePath, File imageFile)
			throws IOException {
		Image img = ImageIO.read(imageFile);
		// 判断图片格式是否正确
		if (img.getWidth(null) == -1) {
			log.info(" can't read,retry!" + "<BR>");
			return;
		} else {
			// 等比缩放
			// 为等比缩放计算输出的图片宽度及高度
			double rate1 = ((double) img.getWidth(null))
					/ (double) Constants.OUTPUT_WIDTH_HTML;
			double rate2 = ((double) img.getHeight(null))
					/ (double) Constants.OUTPUT_HEIGHT_HTML;
			// 根据缩放比率大的进行缩放控制
			double rate = rate1 > rate2 ? rate1 : rate2;
			if (rate > 1.2 || rate < 0.85) {
				rate = 0.98;
			}
			int newWidth = (int) (((double) img.getWidth(null)) / rate);
			int newHeight = (int) (((double) img.getHeight(null)) / rate);
			BufferedImage tag = null;
			/*
			 * Image.SCALE_SMOOTH 的缩略算法 生成缩略图片的平滑度的 优先级比速度高 生成的图片质量比较好 但速度慢
			 */
			FileOutputStream out = new FileOutputStream(savePath);
			if (img.getWidth(null) <= 450) {
				FileInputStream fis = new FileInputStream(imageFile);
				byte[] buffer = new byte[1024];
				int len = 0;
				while ((len = fis.read(buffer)) > 0) {
					out.write(buffer, 0, len);
				}
				fis.close();
				out.close();
				log.info("编辑器中图片，原宽度：" + img.getWidth(null) + "原高度："
						+ img.getHeight(null));
				return;
			} else {
				tag = new BufferedImage(newWidth, newHeight,
						BufferedImage.TYPE_INT_RGB);
				log.info("编辑器中图片，新宽度：" + newWidth + "新高度：" + newHeight);
				tag.getGraphics().drawImage(
						img.getScaledInstance(newWidth, newHeight,
								Image.SCALE_SMOOTH), 0, 0, null);
			}

			// JPEGImageEncoder可适用于其他图片类型的转换
			JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
			encoder.encode(tag);
			out.close();
		}
	}

}
